import os
import pandas as pd


def rename_excel_columns(folder_path):
    # 遍历文件夹中的所有文件
    for file_name in os.listdir(folder_path):
        # 检查文件是否为Excel文件（以.xlsx结尾）
        if file_name.endswith('.xlsx'):
            file_path = os.path.join(folder_path, file_name)

            try:
                # 读取Excel文件
                excel_data = pd.ExcelFile(file_path)

                # 创建一个写入器，用于保存修改后的数据
                with pd.ExcelWriter(file_path, engine='openpyxl') as writer:
                    for sheet_name in excel_data.sheet_names:
                        # 读取每个工作表
                        df = excel_data.parse(sheet_name)

                        # 修改列名
                        if 0 in df.columns:
                            df.rename(columns={0: 'company_name'}, inplace=True)
                        if 2 in df.columns:
                            df.rename(columns={2: 'Products_and_Services'}, inplace=True)

                        # 将修改后的工作表写回文件
                        df.to_excel(writer, sheet_name=sheet_name, index=False)

                print(f"已修改文件: {file_name}")
            except Exception as e:
                print(f"处理文件 {file_name} 时出错: {e}")


# 指定文件夹路径
folder_path = r'./data/base'

# 调用函数
rename_excel_columns(folder_path)